# PolymorphismPersisterBuilder.java

    package org.codefilarete.stalactite.engine.configurer.polymorphism;

    import java.util.HashMap;
    import java.util.Map;
    import java.util.Set;

    import org.codefilarete.reflection.PropertyAccessPoint;
    import org.codefilarete.reflection.ValueAccessPointMap;
    import org.codefilarete.stalactite.dsl.PolymorphismPolicy;
    import org.codefilarete.stalactite.dsl.PolymorphismPolicy.JoinTablePolymorphism;
    import org.codefilarete.stalactite.dsl.PolymorphismPolicy.SingleTablePolymorphism;
    import org.codefilarete.stalactite.dsl.PolymorphismPolicy.TablePerClassPolymorphism;
    import org.codefilarete.stalactite.engine.configurer.AbstractIdentification;
    import org.codefilarete.stalactite.engine.configurer.NamingConfiguration;
    import org.codefilarete.stalactite.engine.configurer.builder.PersisterBuilderContext;
    import org.codefilarete.stalactite.engine.listener.SelectListener;
    import org.codefilarete.stalactite.engine.runtime.AbstractPolymorphismPersister;
    import org.codefilarete.stalactite.engine.runtime.ConfiguredRelationalPersister;
    import org.codefilarete.stalactite.sql.ConnectionConfiguration;
    import org.codefilarete.stalactite.sql.Dialect;
    import org.codefilarete.stalactite.sql.ddl.structure.Column;
    import org.codefilarete.stalactite.sql.ddl.structure.Table;
    import org.codefilarete.stalactite.sql.statement.binder.ColumnBinderRegistry;
    import org.codefilarete.tool.collection.KeepOrderSet;
    import org.codefilarete.tool.exception.NotImplementedException;
    import org.codefilarete.tool.function.Converter;

    /**
     * Builder of polymorphic persisters. Handles {@link PolymorphismPolicy} subtypes, as such, it is the main entry point for polymorphic persisters :
     * it will invoke {@link JoinTablePolymorphismBuilder}, {@link SingleTablePolymorphismBuilder} or {@link TablePerClassPolymorphismBuilder}
     * according to policy. Hence, those builders are not expected to be invoked directly outside this class.
     *
     * @author Guillaume Mary
     */
    public class PolymorphismPersisterBuilder<C, I, T extends Table> implements PolymorphismBuilder<C, I, T> {

    	private final PolymorphismPolicy<C> polymorphismPolicy;
    	private final ConfiguredRelationalPersister<C, I> mainPersister;
    	private final AbstractIdentification<C, I> identification;
    	private final ColumnBinderRegistry columnBinderRegistry;

    	private final Map<PropertyAccessPoint<C, Object>, Column<T, Object>> mainMapping;
    	private final Map<PropertyAccessPoint<C, Object>, Column<T, Object>> mainReadonlyMapping;
    	private final ValueAccessPointMap<C, Converter<Object, Object>, PropertyAccessPoint<C, ?>> mainReadConverters;
    	private final ValueAccessPointMap<C, Converter<Object, Object>, PropertyAccessPoint<C, ?>> mainWriteConverters;
    	private final NamingConfiguration namingConfiguration;
    	private final PersisterBuilderContext persisterBuilderContext;

    	public PolymorphismPersisterBuilder(PolymorphismPolicy<C> polymorphismPolicy,
    										AbstractIdentification<C, I> identification,
    										ConfiguredRelationalPersister<C, I> mainPersister,
    										ColumnBinderRegistry columnBinderRegistry,
    										Map<? extends PropertyAccessPoint<C, Object>, Column<T, Object>> mainMapping,
    										Map<? extends PropertyAccessPoint<C, Object>, Column<T, Object>> mainReadonlyMapping,
    										ValueAccessPointMap<C, ? extends Converter<Object, Object>, PropertyAccessPoint<C, ?>> mainReadConverters,
    										ValueAccessPointMap<C, ? extends Converter<Object, Object>, PropertyAccessPoint<C, ?>> mainWriteConverters,
    										NamingConfiguration namingConfiguration,
    										PersisterBuilderContext persisterBuilderContext) {
    		this.polymorphismPolicy = polymorphismPolicy;
    		this.identification = identification;
    		this.mainPersister = mainPersister;
    		this.columnBinderRegistry = columnBinderRegistry;
    		this.mainMapping = (Map<PropertyAccessPoint<C, Object>, Column<T, Object>>) mainMapping;
    		this.mainReadonlyMapping = (Map<PropertyAccessPoint<C, Object>, Column<T, Object>>) mainReadonlyMapping;
    		this.mainReadConverters = (ValueAccessPointMap<C, Converter<Object, Object>, PropertyAccessPoint<C, ?>>) mainReadConverters;
    		this.mainWriteConverters = (ValueAccessPointMap<C, Converter<Object, Object>, PropertyAccessPoint<C, ?>>) mainWriteConverters;
    		this.namingConfiguration = namingConfiguration;
    		this.persisterBuilderContext = persisterBuilderContext;
    	}

    	@Override
    	public AbstractPolymorphismPersister<C, I> build(Dialect dialect, ConnectionConfiguration connectionConfiguration) {
    		PolymorphismBuilder<C, I, T> polymorphismBuilder;
    		if (polymorphismPolicy instanceof PolymorphismPolicy.SingleTablePolymorphism) {
    			polymorphismBuilder = new SingleTablePolymorphismBuilder<>((SingleTablePolymorphism<C, ?>) polymorphismPolicy,
    					this.identification, this.mainPersister,
    					this.mainMapping, this.mainReadonlyMapping,
    					this.mainReadConverters, this.mainWriteConverters,
    					this.columnBinderRegistry, this.namingConfiguration,
    					this.persisterBuilderContext);
    		} else if (polymorphismPolicy instanceof PolymorphismPolicy.TablePerClassPolymorphism) {
    			polymorphismBuilder = new TablePerClassPolymorphismBuilder<>((TablePerClassPolymorphism<C>) polymorphismPolicy,
    					this.identification, this.mainPersister,
    					this.mainMapping, this.mainReadonlyMapping,
    					this.mainReadConverters, this.mainWriteConverters,
    					this.columnBinderRegistry, this.namingConfiguration,
    					this.persisterBuilderContext);
    		} else if (polymorphismPolicy instanceof PolymorphismPolicy.JoinTablePolymorphism) {
    			polymorphismBuilder = new JoinTablePolymorphismBuilder<>((JoinTablePolymorphism<C>) polymorphismPolicy,
    					this.identification, this.mainPersister,
    					this.columnBinderRegistry, this.namingConfiguration,
    					this.persisterBuilderContext);
    		} else {
    			// this exception is more to satisfy Sonar than for real case
    			throw new NotImplementedException("Given policy is not implemented : " + polymorphismPolicy);
    		}
    		AbstractPolymorphismPersister<C, I> result = polymorphismBuilder.build(dialect, connectionConfiguration);
    		// We transfer listeners so that all actions are made in the same "event listener context" : all listeners are aggregated in a top level one.
    		// Made in particular for relation cascade triggering.
    		// TODO: move main persister selectListener propagation to sub-persisters here ? see JoinTablePolymorphismBuilder
    		mainPersister.getPersisterListener().moveTo(result.getPersisterListener());

    		// We propagate listeners to sub-persisters to make them trigger their own select listeners, in particular the relation ones and their
    		// already-assigned persisted flag. This is necessary because the main persister is not aware of sub-relations.
    		result.addSelectListener(new SelectListenerPropagator<>(result));

    		return result;
    	}

    	private static class SelectListenerPropagator<C, I> implements SelectListener<C, I> {

    		private final AbstractPolymorphismPersister<C, I> parentPersister;

    		private SelectListenerPropagator(AbstractPolymorphismPersister<C, I> parentPersister) {
    			this.parentPersister = parentPersister;
    		}

    		@Override
    		public void beforeSelect(Iterable<I> ids) {
    			parentPersister.getSubEntitiesPersisters().values()
    					.forEach(subPersister -> subPersister.getPersisterListener().getSelectListener().beforeSelect(ids));
    		}

    		@Override
    		public void afterSelect(Set<? extends C> entities) {
    			Map<Class<C>, Set<C>> entitiesPerType = computeEntitiesPerType(entities);
    			parentPersister.getSubEntitiesPersisters().forEach((type, subPersister) -> {
    				Set<C> subEntities = entitiesPerType.get(type);
    				if (subEntities != null) {
    					subPersister.getPersisterListener().getSelectListener().afterSelect(subEntities);
    				}
    			});
    		}

    		@Override
    		public void onSelectError(Iterable<I> ids, RuntimeException exception) {
    			// since ids are not those of its entities, we should not pass them as argument
    			parentPersister.getSubEntitiesPersisters().values()
    					.forEach(subPersister -> subPersister.getPersisterListener().getSelectListener().onSelectError(ids, exception));
    		}

    		private Map<Class<C>, Set<C>> computeEntitiesPerType(Iterable<? extends C> entities) {
    			Map<Class<C>, Set<C>> entitiesPerType = new HashMap<>();
    			entities.forEach(entity ->
    					entitiesPerType.computeIfAbsent((Class<C>) entity.getClass(), p -> new KeepOrderSet<>()).add(entity)
    			);
    			return entitiesPerType;
    		}
    	}
    }